解答1[Java]:
class Solution {
public int removeDuplicates(int[] nums) {
int i = 0;
for (int n : nums)
if (i < 2 || n > nums[i - 2])
nums[i++] = n;
return i;
}
}
补充解答:修改版
int removeDuplicates(vector<int>& nums,int k) {
if(nums.size()<k) return nums.size(); // if array size is less than k then return the same
int i,j;
for(i=k,j=k ; i<nums.size();i++)
if(nums[j-k]!=nums[i]) nums[j++]=nums[i];
return j;
}
这个解答,可以适用于保留最多不超过 K 个相同的元素。
和第一种解法的思路相似,j-k
就是 lookback,向后看。第一种解法是 nums[i-2]
,往后看两位。现在是往后看 k 位。